package com.lagou.mall.esflow.server.provider;

import com.alibaba.fastjson.JSONObject;
import com.alibaba.nacos.api.config.annotation.NacosValue;
import com.lagou.mall.esflow.client.Binlog;
import com.lagou.mall.esflow.server.config.EnvConfig;
import java.util.Set;
import javax.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

@Component
@Slf4j
public class FilterOperatorData {

    @Resource
    private EnvConfig envConfig;

    // pre 环境只操作字段 key 为 value 的
    @NacosValue(value = "${pre.filter.key:}", autoRefreshed = true)
    private String preFilterKey;
    @NacosValue(value = "${pre.filter.value:}", autoRefreshed = true)
    private Set<String> preFilterValue;


    /**
     * pre环境只写指定数据
     */
    public boolean filter(Binlog binlog) {
        JSONObject json = binlog.getData();
        if (envConfig.isPre()) {
            // 预发环境只处理该数据
            if (json.containsKey(preFilterKey) && preFilterValue.contains(json.getString(preFilterKey))) {
                return false;
            }
            // 其他数据均不处理
            log.info("当前环境不处理该数据");
            return true;
        } else if (envConfig.isProd()) {
            // 生产环境不处理该数据
            if (json.containsKey(preFilterKey) && preFilterValue.contains(json.getString(preFilterKey))) {
                log.info("当前环境不处理该数据");
                return true;
            }
            // 其他数据均处理
            return false;
        }
        // 其他环境
        return false;
    }
}